//*********************************
package cn.zhuangcloud.newversion.common.model;

import cn.zhuangcloud.newversion.common.Config;
import com.jfinal.kit.PathKit;
import com.jfinal.plugin.activerecord.dialect.PostgreSqlDialect;
import com.jfinal.plugin.activerecord.generator.Generator;
import com.jfinal.plugin.activerecord.generator.MetaBuilder;
import com.jfinal.plugin.druid.DruidPlugin;
//*********************************

import javax.sql.DataSource;

public class _Generator {

    //*********************************
    private static DataSource getDataSource() {
        DruidPlugin druidPlugin = Config.getDruidPlugin();
        druidPlugin.start();
        return druidPlugin.getDataSource();
    }

    public static void main(String[] args) {
        // base model 所使用的包名*********************************
        String baseModelPackageName = "cn.ssdut153.background.common.model.base";
        // base model 文件保存路径*********************************
        String baseModelOutputDir = PathKit.getWebRootPath() + "/../src/main/java/cn/ssdut153/background/common/model/base";

        // model 所使用的包名 (MappingKit 默认使用的包名)*********************************
        String modelPackageName = "cn.ssdut153.background.common.model";
        // model 文件保存路径 (MappingKit 与 DataDictionary 文件默认保存路径)
        String modelOutputDir = baseModelOutputDir + "/..";

        // 创建生成器
        DataSource dp = getDataSource();
        Generator generator = new Generator(dp, baseModelPackageName, baseModelOutputDir, modelPackageName, modelOutputDir);
        // 设置是否生成链式 setter 方法
        generator.setGenerateChainSetter(false);
        //*********************************
        //generator.setMetaBuilder(new SqlServerMetaBuilder(dp));
        generator.setMetaBuilder(new MetaBuilder(dp));
        // 添加不需要生成的表名*********************************
        //generator.addExcludedTable("153_1_1", "153code", "153discuss", "153gradesubmit", "153msg", "153settings", "153web", "153wechat", "Table", "User");
        // 设置是否在 Model 中生成 dao 对象
        generator.setGenerateDaoInModel(false);
        //*********************************
        //generator.setDialect(new SqlServerDialect());
        generator.setDialect(new PostgreSqlDialect());
        // 设置是否生成链式 setter 方法
        generator.setGenerateChainSetter(true);
        // 设置是否生成字典文件
        generator.setGenerateDataDictionary(false);
        // 设置需要被移除的表名前缀用于生成modelName。例如表名 "osc_user"，移除前缀 "osc_"后生成的model名为 "User"而非 OscUser
//        generator.setRemovedTableNamePrefixes("153");
        // 生成
        generator.generate();
    }
}
